package com.etcn.common.utils.datacompress;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.zip.GZIPInputStream;
import java.util.zip.GZIPOutputStream;

/**
 * GZIP压缩工具类
 * @author guozc
 * @date: 2016-08-02 13:28:55
 */
public class GZIPUtil {

	 /**
     * 字符串的压缩
     * @param str 待压缩的字符串
     * @throws IOException
     */
    public static String compress(String str) throws Exception {
        if (null == str || str.length() <= 0) {
            return str;
        }
        // 创建一个新的 byte 数组输出流
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        // 使用默认缓冲区大小创建新的输出流
        GZIPOutputStream gzip = new GZIPOutputStream(out);
        // 将 b.length 个字节写入此输出流
        gzip.write(str.getBytes());
        gzip.close();
        // 使用指定的 charsetName，通过解码字节将缓冲区内容转换为字符串
        return out.toString("ISO-8859-1");
    }
    
    /**
     * 对byte[]进行压缩
     * @param data 待压缩的
     * @return 压缩后的
     * @throws IOException
     */
    public static byte[] compress(byte[] data) throws Exception {
        if (data == null) {
        	return data;
        }
        System.out.println("before:" + data.length);
        byte[] newData = null ; 
        // 创建一个新的 byte 数组输出流
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        // 使用默认缓冲区大小创建新的输出流
        GZIPOutputStream gzip = new GZIPOutputStream(out);
        // 将 b.length 个字节写入此输出流
        gzip.write(data);
        gzip.close();
        newData = out.toByteArray() ; 
        System.out.println("after:" + newData.length);  
        return newData;
    }
     
    /**
     * 字符串的解压
     * @param str 对字符串解压
     * @return 返回解压缩后的字符串
     * @throws Exception
     */
    public static String unCompress(String str) throws Exception {
        if (null == str || str.length() <= 0) {
            return str;
        }
        // 创建一个新的 byte 数组输出流
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        // 创建一个 ByteArrayInputStream，使用 buf 作为其缓冲区数组
        ByteArrayInputStream in = new ByteArrayInputStream(str
                .getBytes("ISO-8859-1"));
        // 使用默认缓冲区大小创建新的输入流
        GZIPInputStream gzip = new GZIPInputStream(in);
        byte[] buffer = new byte[256];
        int n = 0;
        while ((n = gzip.read(buffer)) >= 0) {// 将未压缩数据读入字节数组
            // 将指定 byte 数组中从偏移量 off 开始的 len 个字节写入此 byte数组输出流
            out.write(buffer, 0, n);
        }
        // 使用指定的 charsetName，通过解码字节将缓冲区内容转换为字符串
        return out.toString("utf-8");
    }
    
    /**
     * 输入流的解压
     * @param str 对字符串解压
     * @return 返回从流中读取后并且解压缩后的字符串
     * @throws Exception
     */
    public static String unCompress(InputStream in) throws Exception {
        // 创建一个新的 byte 数组输出流
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        // 使用默认缓冲区大小创建新的输入流
        GZIPInputStream gzip = new GZIPInputStream(in);
        byte[] buffer = new byte[256];
        int n = 0;
        while ((n = gzip.read(buffer)) >= 0) {// 将未压缩数据读入字节数组
            // 将指定 byte 数组中从偏移量 off 开始的 len 个字节写入此 byte数组输出流
            out.write(buffer, 0, n);
        }
        // 使用指定的 charsetName，通过解码字节将缓冲区内容转换为字符串
        return out.toString("utf-8");
    }
    
    /**
     * byte[]的解压
     * @param 未解压前
     * @return 解压后
     * @throws Exception
     */
    public static byte[] unCompress(byte[] data) throws Exception {
        if (data == null) {
            return data;
        }
        System.out.println("before:" + data.length);
        byte[] newData = null ; 
        // 创建一个新的 byte 数组输出流
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        // 创建一个 ByteArrayInputStream，使用 buf 作为其缓冲区数组
        ByteArrayInputStream in = new ByteArrayInputStream(data);
        // 使用默认缓冲区大小创建新的输入流
        GZIPInputStream gzip = new GZIPInputStream(in);
        byte[] buffer = new byte[256];
        int n = 0;
        while ((n = gzip.read(buffer)) >= 0) {// 将未压缩数据读入字节数组
            // 将指定 byte 数组中从偏移量 off 开始的 len 个字节写入此 byte数组输出流
            out.write(buffer, 0, n);
        }
        newData = out.toByteArray();
        System.out.println("after:" + newData.length);  
        // 使用指定的 charsetName，通过解码字节将缓冲区内容转换为字符串
        return newData;
    }
    
}
